package com.sheng.leetcode.year2022.month09.day22;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * @author liusheng
 * @date 2022/09/22
 *<p>
 * 1640. 能否连接形成数组<p>
 *<p>
 * 给你一个整数数组 arr ，数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces，其中的整数也 互不相同 。<p>
 * 请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是，不允许 对每个数组 pieces[i] 中的整数重新排序。<p>
 * 如果可以连接 pieces 中的数组形成 arr ，返回 true ；否则，返回 false 。<p>
 *<p>
 * 示例 1：<p>
 * 输入：arr = [15,88], pieces = [[88],[15]]<p>
 * 输出：true<p>
 * 解释：依次连接 [15] 和 [88]<p>
 *<p>
 * 示例 2：<p>
 * 输入：arr = [49,18,16], pieces = [[16,18,49]]<p>
 * 输出：false<p>
 * 解释：即便数字相符，也不能重新排列 pieces[0]<p>
 *<p>
 * 示例 3：<p>
 * 输入：arr = [91,4,64,78], pieces = [[78],[4,64],[91]]<p>
 * 输出：true<p>
 * 解释：依次连接 [91]、[4,64] 和 [78]<p>
 *<p>
 * 提示：<p>
 *<p>
 * 1 <= pieces.length <= arr.length <= 100<p>
 * sum(pieces[i].length) == arr.length<p>
 * 1 <= pieces[i].length <= arr.length<p>
 * 1 <= arr[i], pieces[i][j] <= 100<p>
 * arr 中的整数 互不相同<p>
 * pieces 中的整数 互不相同（也就是说，如果将 pieces 扁平化成一维数组，数组中的所有整数互不相同）<p>
 *<p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/check-array-formation-through-concatenation">...</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1640 {

    @Test
    public void test01() {
//        int[] arr = {15,88};
//        int[][] pieces = {{15}, {88}};
//        int[] arr = {49,18,16};
//        int[][] pieces = {{16,18,49}};
        int[] arr = {91,4,64,78};
        int[][] pieces = {{78},{4,64},{91}};
        System.out.println(new Solution().canFormArray(arr, pieces));
    }
}
class Solution {
    public boolean canFormArray(int[] arr, int[][] pieces) {
        Map<Integer, Integer> map = new HashMap<>(16);
        int i = 0;
        while (i < arr.length) {
            boolean flag = false;
            for (int j = 0; j < pieces.length; j++) {
                int k = 0;
                // 如果找到了符合的数组
                if (pieces[j][k] == arr[i] && !map.containsKey(j)) {
                    // 其中的每个元素都要相等
                    while (k < pieces[j].length) {
                        if (pieces[j][k] == arr[i]) {
                            k++;
                            i++;
                        } else {
                            return false;
                        }
                    }
                    flag = true;
                    // 将这个数组设置为已被使用
                    map.put(j, 0);
                    break;
                }
            }
            // 如果在数组pieces中没有找到符合arr[i]的数组，直接返回false
            if (!flag) {
                return flag;
            }
        }
        return true;
    }
}
